package io.renren.controller;

import io.renren.annotation.Login;
import io.renren.common.exception.RenException;
import io.renren.common.utils.Result;
import io.renren.common.validator.ValidatorUtils;
import io.renren.common.validator.group.AddGroup;
import io.renren.common.validator.group.DefaultGroup;
import io.renren.modules.health.dto.FoodDTO;
import io.renren.modules.health.dto.FoodPlanDTO;
import io.renren.modules.health.dto.FoodPlanDetailDTO;
import io.renren.modules.health.entity.FoodPlanDetailEntity;
import io.renren.modules.health.service.FoodPlanDetailService;
import io.renren.modules.health.service.FoodPlanService;
import io.renren.modules.health.service.FoodService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import springfox.documentation.annotations.ApiIgnore;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;


/**
 * 饮食方案
 */
@RestController
@RequestMapping("api/foodplan")
@Api(tags="饮食方案")
public class FoodPlanController {
    @Autowired
    private FoodPlanService foodPlanService;
    @Autowired
    private FoodPlanDetailService foodPlanDetailService;
    @Autowired
    private FoodService foodService;

    @Login
    @GetMapping("list")
    @ApiOperation("查询饮食计划")
    @ApiImplicitParams({
        @ApiImplicitParam(name = "cusId", value = "客户id", paramType = "query", required = true, dataType="int") ,
    })
    public Result<List<FoodPlanDTO>> page(@ApiIgnore @RequestParam Map<String, Object> params){
        foodPlanService.genCustomerFoodPlan(Long.parseLong((String) params.get("cusId")), false);
        List<FoodPlanDTO> foodPlanDTOS = foodPlanService.queryFoodPlanWithDetail(Long.parseLong((String) params.get("cusId")));
        return new Result<List<FoodPlanDTO>>().ok(foodPlanDTOS);
    }

    @Login
    @GetMapping("{id}")
    @ApiOperation("信息")
    public Result<FoodPlanDTO> get(@PathVariable("id") Long id){
        FoodPlanDTO data = foodPlanService.get(id);
        HashMap<String, Object> param = new HashMap<>();
        param.put("foodPlanId",id);
        List<FoodPlanDetailDTO> foodPlanDetailDTOS = foodPlanDetailService.list(param);
        param.put("foodIds",foodPlanDetailDTOS.stream().map(FoodPlanDetailDTO::getFoodId).distinct().collect(Collectors.toList()));
        data.setFoodPlanDetails(foodPlanDetailDTOS);
        List<FoodDTO> foodDTOS = foodService.list(param);
        foodPlanDetailDTOS.forEach(foodPlanDetail -> foodPlanDetail.setFoodName(
                    foodDTOS.stream().filter(food -> foodPlanDetail.getFoodId().equals(food.getId()))
                            .findFirst()
                            .orElse(new FoodDTO()).getFoodName()));

        return new Result<FoodPlanDTO>().ok(data);
    }

    @PostMapping
    @Login
    @ApiOperation("保存")
    public Result save(@RequestBody FoodPlanDTO dto){
        //效验数据
        ValidatorUtils.validateEntity(dto, AddGroup.class, DefaultGroup.class);

        foodPlanService.save(dto);
        if (!Objects.isNull(dto.getFoodPlanDetails()) && !dto.getFoodPlanDetails().isEmpty()) {
            List<FoodPlanDetailEntity> foodPlanDetailEntities = dto.getFoodPlanDetails().stream()
                    .map(fpd -> {
                        fpd.setFoodPlanId(dto.getId());
                        FoodPlanDetailEntity foodPlanEntity = new FoodPlanDetailEntity();
                        BeanUtils.copyProperties(fpd, foodPlanEntity);
                        return foodPlanEntity;
                    })
                    .collect(Collectors.toList());
            foodPlanDetailService.insertBatch(foodPlanDetailEntities);
        } else throw new RenException(100,"饮食方案未添加食品");
        return new Result();
    }

}